from pymongo import MongoClient
import os
import json
import pymongo
client = MongoClient(port=27017)
db=client['tiktok_p2']
obj={
    '_id':'',
    'frame_level':{'f0':{'text':{},'img':{},'audio':{}},
                   'f1':{'text':{},'img':{},'audio':{}},
                   'f2':{'text':{},'img':{},'audio':{}},
                   'f3':{'text':{},'img':{},'audio':{}},
                   'f4':{'text':{},'img':{},'audio':{}},
                   'f5':{'text':{},'img':{},'audio':{}},
                   'f6':{'text':{},'img':{},'audio':{}},
                   'f7':{'text':{},'img':{},'audio':{}},
                   'f8':{'text':{},'img':{},'audio':{}},
                   'f9':{'text':{},'img':{},'audio':{}}
    },
    'video_level':{
    }
}
# add=[]
# for ht in os.listdir('E:\\tiktok_new\\video\\'):
#     add.append(ht)
add=['thriftshop']
for hashtag in add:
    i=0
    scss=0
    fl=0
    hashtag=hashtag.replace('#','').lower()
    for fname in os.listdir('E:\\tiktok_new\\video\\'+hashtag+'\\'):
        id=fname.replace('.mp4','')
        temp=obj
        temp['_id']=id
        try:
            db[hashtag].insert_one(temp)
            scss+=1
        except pymongo.errors.DuplicateKeyError:
            fl+=1
        i+=1
    print(hashtag,scss,fl)

